import random
import pandas as pd

# 定义基础参数池
M_pool = [1, 128, 256, 384, 512]  # M可选值
N_pool = [1024, 2048, 4096, 5120, 6656, 8192]  # N可选值
dtype_pool = ["float16", "float32"]  # 数据类型可选值
total_groups = 100  # 目标总组数
random.seed(42)  # 固定种子确保可复现（删除则每次随机）

# 生成完全随机组合：每次独立抽取M、N、dtype
random_combinations = []
for _ in range(total_groups):
    sigtype = random.choice(dtype_pool)  # 随机选数据类型
    m = random.choice(M_pool)  # 随机选M值
    n = random.choice(N_pool)  # 随机选N值
    random_combinations.append({"sigtype": sigtype, "m": m, "n": n})

# 转换为DataFrame并保存
df = pd.DataFrame(random_combinations)
output_path = "./params/silu_performance_data.csv"
df.to_csv(output_path, index=False, encoding="utf-8")